diff --git a/benchmarks/lightning/main.py b/benchmarks/lightning/main.py
index b5398478..9295c504 100644
--- a/benchmarks/lightning/main.py
+++ b/benchmarks/lightning/main.py
@@ -10,13 +10,6 @@
 import lightning as L
 import torchvision.models as torchvision_models
 
-# HPU fix
-os.environ["RANK"] = os.getenv("RANK","-1")
-os.environ["LOCAL_RANK"] = os.getenv("LOCAL_RANK","-1")
-os.environ["WORLD_SIZE"] = os.getenv("WORLD_SIZE", "1")
-os.environ["LOCAL_WORLD_SIZE"] = os.getenv("LOCAL_WORLD_SIZE", "1")
-
-
 from benchmate.dataloader import imagenet_dataloader, dataloader_arguments
 
 
@@ -89,8 +82,6 @@ def main():
     observer, monitor = prepare_voir()
     loader = observer.loader(imagenet_dataloader(args, model, rank, world_size))
 
-    # ValueError: The `HPUAccelerator` can only be used with a `SingleHPUStrategy` or `HPUParallelStrategy`, found DDPStrategy.
-    
     # train model
     trainer = L.Trainer(
         accelerator="auto", 
diff --git a/docker/Makefile b/docker/Makefile
index 46789d70..d0c77409 100644
--- a/docker/Makefile
+++ b/docker/Makefile
@@ -3,7 +3,7 @@
 
 # bench = rlhf-single
 bench = "lightning"
-lazy = 0
+lazy = 1
 
 hpu:
 	git add --all