diff --git a/lbmpy/max_domain_size_info.py b/lbmpy/max_domain_size_info.py
index fcd2ed0984cdcb54600bc63c1d29651a28f3357c..3e6b961525bf58c5279ed37ae6111a4b2c81c462 100644
--- a/lbmpy/max_domain_size_info.py
+++ b/lbmpy/max_domain_size_info.py
@@ -41,6 +41,15 @@ except ImportError:
     cupy = None
     device = None
 
+if cupy:
+    try:
+        device = cupy.cuda.Device(pystencils.GPU_DEVICE)
+    except AttributeError:
+        warnings.warn("You are using an old pystencils version. Consider updating it")
+        device = cupy.cuda.Device(0)
+else:
+    device = None
+
 try:
     from psutil import virtual_memory
 except ImportError: