diff --git a/arm_1dof/bullet_arm_1dof.py b/arm_1dof/bullet_arm_1dof.py
index 5291446..40ab1db 100644
--- a/arm_1dof/bullet_arm_1dof.py
+++ b/arm_1dof/bullet_arm_1dof.py
@@ -1,6 +1,6 @@
+import arm_1dof.robot_arm_1dof
+import pybullet
 from tokenize import String
-import pybullet 
-import robot_arm_1dof
 
 import os
 
diff --git a/arm_1dof/test_arm_1dof.py b/arm_1dof/test_arm_1dof.py
index 7407b4c..5dc973d 100755
--- a/arm_1dof/test_arm_1dof.py
+++ b/arm_1dof/test_arm_1dof.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 
-from bullet_arm_1dof import BulletArm1Dof
-from robot_arm_1dof import RobotArm1Dof
+from arm_1dof.bullet_arm_1dof import BulletArm1Dof
+from arm_1dof.robot_arm_1dof import RobotArm1Dof
 
 import pybullet