diff --git a/installsystems/image.py b/installsystems/image.py
index ed58cb7596a0c745f2dd662d4c3cf82c85a9f770..389ebef6b95568c5f75813059f746666dd36838b 100644
--- a/installsystems/image.py
+++ b/installsystems/image.py
@@ -318,6 +318,7 @@ class PackageImage(Image):
 
     def run_setup(self, gl):
         '''Run setup scripts'''
+        gl["image"] = self
         self.run_scripts(gl, "setup")
 
     def run_scripts(self, gl, directory):