diff --git a/pungi/wrappers/kojiwrapper.py b/pungi/wrappers/kojiwrapper.py
index de405815..4348f884 100644
--- a/pungi/wrappers/kojiwrapper.py
+++ b/pungi/wrappers/kojiwrapper.py
@@ -869,6 +869,7 @@ class KojiMockWrapper(object):
     lock = threading.Lock()
 
     def __init__(self, compose):
+        self.compose = compose
         try:
             self.profile = self.compose.conf["koji_profile"]
         except KeyError: