diff --git a/drive_between_arucos.py b/drive_between_arucos.py
index 62679d6..12f12d1 100644
--- a/drive_between_arucos.py
+++ b/drive_between_arucos.py
@@ -64,7 +64,7 @@ def find_arucos(arlo):
             print(rot, value)
             aruco_dict[key] = np.dot(rot, value)
 
-    return list(aruco_dict.values())[:2]
+    return np.array(aruco_dict.values())[:2]
 
 def main():
     arlo = Robot()