diff --git a/script/power-supply.py b/script/power-supply.py
index b18a254..b8e2d91 100644
--- a/script/power-supply.py
+++ b/script/power-supply.py
@@ -1,6 +1,3 @@
-#Script imported from: PO2DV-NCS-HW_ps.xml
-from mathutils import estimate_peak_indexes, fit_gaussians, create_fit_point_list
-
 #Pre-actions
 caput('PO2DV-NCS-VHQ1:Set-RampA', '10')
 sleep(0.1)
@@ -64,4 +61,4 @@ ActualIA.close()
 
 scan.end()
 
-return 1
\ No newline at end of file
+sys.exit(1)
\ No newline at end of file