diff --git a/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py b/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py
index 3f6c5ddd..0cea9631 100644
--- a/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py
+++ b/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py
@@ -63,7 +63,8 @@ class NLinkArm:
 
         if plot:
             self.fig = plt.figure()
-            self.ax = Axes3D(self.fig)
+            self.ax = Axes3D(self.fig, auto_add_to_figure=False)
+            self.fig.add_axes(self.ax)
 
             x_list = []
             y_list = []
diff --git a/ArmNavigation/n_joint_arm_to_point_control/n_joint_arm_to_point_control.py b/ArmNavigation/n_joint_arm_to_point_control/n_joint_arm_to_point_control.py
index b7834ee9..36093df0 100644
--- a/ArmNavigation/n_joint_arm_to_point_control/n_joint_arm_to_point_control.py
+++ b/ArmNavigation/n_joint_arm_to_point_control/n_joint_arm_to_point_control.py
@@ -4,7 +4,11 @@ Inverse kinematics for an n-link arm using the Jacobian inverse method
 Author: Daniel Ingram (daniel-s-ingram)
         Atsushi Sakai (@Atsushi_twi)
 """
+
 import numpy as np
+import sys
+from pathlib import Path
+sys.path.append(str(Path(__file__).parent.parent.parent))
 
 from ArmNavigation.n_joint_arm_to_point_control.NLinkArm import NLinkArm