diff --git a/image_generation/Storyteller-Studio-Filter/main.py b/image_generation/Storyteller-Studio-Filter/main.py
index 90aca7c..c466493 100644
--- a/image_generation/Storyteller-Studio-Filter/main.py
+++ b/image_generation/Storyteller-Studio-Filter/main.py
@@ -150,10 +150,10 @@ def main():
     parser.add_argument("--samplers", type=str, required=False, help="Text input called samplers", choices=list(_au111_to_diffusers_samplers.keys()))
        
     # # ZDisket: We shoud really keep these static and instead use an upsampler afterwards instead
-    parser.add_argument("--width", required=False, help="Width value (power of 2, between 256 and 4096)")
-    parser.add_argument("--height", required=False, help="Height value (power of 2, between 256 and 4096)")
+    parser.add_argument("--width", type=int, required=False, help="Width value (power of 2, between 256 and 4096)")
+    parser.add_argument("--height", type=int,required=False, help="Height value (power of 2, between 256 and 4096)")
     
-    parser.add_argument("--cfg-scale", type=InputCheckers.check_cfg_scale, required=True, help="Integer input called CFG scale (range from 1-30)")
+    parser.add_argument("--cfg-scale",type=float,required=True, help="Integer input called CFG scale (range from 1-30 can be .5)")
     parser.add_argument("--seed", type=int, required=False, help="Seed value (integer of any value)")
     parser.add_argument("--loRA-path", type=InputCheckers.valid_path, required=False, help="Path object to the loRA")
     parser.add_argument("--check-point", type=InputCheckers.valid_path, required=True, help="Path object to the Check Point")
diff --git a/image_generation/Storyteller-Studio-Filter/test.sh b/image_generation/Storyteller-Studio-Filter/test.sh
index 6f297dc..32e7b8d 100755
--- a/image_generation/Storyteller-Studio-Filter/test.sh
+++ b/image_generation/Storyteller-Studio-Filter/test.sh
@@ -5,7 +5,7 @@ python3.10 main.py \
 --samplers "DPM++ 2M SDE Heun Karras" \
 --width 1024 \
 --height 1024 \
---cfg-scale 8 \
+--cfg-scale 8.5 \
 --seed 1 \
 --loRA-path "xs4vow.safetensors" \
 --check-point "waifuDiffusionBeta03_beta3.safetensors" \