diff --git a/matrixai/__init__.py b/matrixai/__init__.py
index a6cfbac..79505a3 100644
--- a/matrixai/__init__.py
+++ b/matrixai/__init__.py
@@ -36,7 +36,7 @@ import uuid
 import subprocess
 
 from pyimagine import Imagine
-from pyimagine.constants import Inspiration
+from pyimagine.constants import Inspiration, Style, Ratio
 
 def run():
     if not USERNAME or not SERVER or not PASSWORD:
@@ -107,8 +107,8 @@ def run():
                 imagine = Imagine()
 
                 filename = str(uuid.uuid4()) + ".png"
-                style_enum = Style[style]
-                ratio_enum = Ratio[ratio]
+                # style_enum = Style[style]
+                # ratio_enum = Ratio[ratio]
 
                 img_data = await imagine.sdprem(
                     prompt=prompt,