diff --git a/wolverine.py b/wolverine.py
index e9d12d6..2e96b08 100644
--- a/wolverine.py
+++ b/wolverine.py
@@ -23,10 +23,18 @@ with open("prompt.txt") as f:
 
 def run_script(script_name, script_args):
     script_args = [str(arg) for arg in script_args]
+    """
+    If script_name.endswith(".py") then run with python
+    else run with node
+    """
+    subprocess_args = (
+        [sys.executable, script_name, *script_args]
+        if script_name.endswith(".py")
+        else ["node", script_name, *script_args]
+    )
+
     try:
-        result = subprocess.check_output(
-            [sys.executable, script_name, *script_args], stderr=subprocess.STDOUT
-        )
+        result = subprocess.check_output(subprocess_args, stderr=subprocess.STDOUT)
     except subprocess.CalledProcessError as e:
         return e.output.decode("utf-8"), e.returncode
     return result.decode("utf-8"), 0
@@ -173,12 +181,7 @@ def main(script_name, *script_args, revert=False, model=DEFAULT_MODEL):
     shutil.copy(script_name, script_name + ".bak")
 
     while True:
-        output = ""
-        returncode = 0
-        if script_name.endswith(".js"):
-            output, returncode = run_node(script_name, script_args)
-        else:
-            output, returncode = run_script(script_name, script_args)
+        output, returncode = run_script(script_name, script_args)
 
         if returncode == 0:
             cprint("Script ran successfully.", "blue")