diff --git a/examples/pybullet/vr_kuka_control.py b/examples/pybullet/vr_kuka_control.py
index ba52b5c18..dccbf5453 100644
--- a/examples/pybullet/vr_kuka_control.py
+++ b/examples/pybullet/vr_kuka_control.py
@@ -2,6 +2,8 @@
 # Require p.setInternalSimFlags(0) in kuka_setup
 import pybullet as p
 import math
+import numpy as np
+
 p.connect(p.SHARED_MEMORY)
 
 kuka = 3