diff --git a/src/vector.js b/src/vector.js
index d62dd66..86ac65b 100644
--- a/src/vector.js
+++ b/src/vector.js
@@ -76,9 +76,9 @@ function vectorLimit(self, value) {
 function vectorClass($gbl, $loc) {
     $loc.__init__ = makeFunc(vectorInit, "__init__", [
         self,
-        { "x": int_, optional },
-        { "y": int_, optional },
-        { "z": int_, optional }
+        { "x": [ int_, float_ ], optional },
+        { "y": [ int_, float_ ], optional },
+        { "z": [ int_, float_ ], optional }
     ]);
 
     $loc.__getattr__ = new Sk.builtin.func(function (self, key) {
@@ -96,9 +96,9 @@ function vectorClass($gbl, $loc) {
 
     $loc.set = makeFunc(vectorSet, "set", [
         self,
-        { "x": int_ },
-        { "x": int_, optional },
-        { "x": int_, optional }
+        { "x": [ int_, float_ ] },
+        { "y": [ int_, float_ ], optional },
+        { "z": [ int_, float_ ], optional }
     ]);
 
     $loc.mag = makeFunc(self => self.v.mag(), "mag", [ self ]);
@@ -157,4 +157,4 @@ function vectorClass($gbl, $loc) {
     $loc.__repr__ = makeFunc(self => self.v.toString(), "repr", [ self ]);
 }
 
-export default mod => buildClass(mod, vectorClass, "PVector", []);
\ No newline at end of file
+export default mod => buildClass(mod, vectorClass, "PVector", []);