diff --git a/image_ref/main_sweep.py b/image_ref/main_sweep.py
index 96d46f2b21e5540c0738d6aa08a7970ace868798..f7c47301881348d0fc1fba00cacc5df9dcdd4eb5 100644
--- a/image_ref/main_sweep.py
+++ b/image_ref/main_sweep.py
@@ -15,7 +15,7 @@ if __name__ == '__main__':
             "batch_size": {"value": 64},
             "positive_prop": {"distribution" : "uniform","max": 95., "min": 5.},
             "opti": {"value": "adam"},
-            "model": {"value": "resnet18"},
+            "model": {"value": "ResNet18"},
             "sampler": {"values": ["random","balanced"]},
             "dataset_train_dir": {"value": "data/processed_data_wiff/npy_image/train_data"},
             "dataset_val_dir": {"value": "data/processed_data_wiff/npy_image/test_data"},