diff --git a/docker/Makefile b/docker/Makefile
index 6826d1b3..4c026af1 100644
--- a/docker/Makefile
+++ b/docker/Makefile
@@ -2,7 +2,7 @@
 
 
 # bench = rlhf-single
-bench = "lightning-gpus"
+bench = "lightning"
 lazy = 0
 
 hpu: