diff --git a/test/test_energy.py b/test/test_energy.py
index 7b8459e1e7b270cf66d77b2fa56ec4bc3df9f978..56f1531c4a1293f86af81f77e7cdfd622a193f4c 100644
--- a/test/test_energy.py
+++ b/test/test_energy.py
@@ -1,3 +1,7 @@
+import json
+
+from helper import *
+
 def energy_model_instance():
     return {
         "power_sleep":9.75,