diff --git a/gprMax/config.py b/gprMax/config.py
index 253a08dc..7150e7db 100644
--- a/gprMax/config.py
+++ b/gprMax/config.py
@@ -90,8 +90,9 @@ class ModelConfig:
             except:
                 deviceID = 0
 
-            self.device = {"dev": sim_config.set_model_device(deviceID), "snapsgpu2cpu": False}
-            self.device["deviceID"] = deviceID
+            self.device = {"dev": sim_config.set_model_device(deviceID), 
+                           "deviceID": deviceID,
+                           "snapsgpu2cpu": False}
 
         # Total memory usage for all grids in the model. Starts with 50MB overhead.
         self.mem_overhead = 65e6