diff --git a/skynet/dgpu/daemon.py b/skynet/dgpu/daemon.py
index d4fecf3..220fe3c 100644
--- a/skynet/dgpu/daemon.py
+++ b/skynet/dgpu/daemon.py
@@ -171,7 +171,7 @@ class SkynetDGPUDaemon:
                 logging.info(f'working on {body}')
 
                 resp = await self.conn.begin_work(rid)
-                if 'code' in resp:
+                if not resp or 'code' in resp:
                     logging.info(f'probably being worked on already... skip.')
 
                 else:
diff --git a/skynet/dgpu/pipes/flux_inpaint.py b/skynet/dgpu/pipes/flux_inpaint.py
index ba0b9e7..2f88d58 100644
--- a/skynet/dgpu/pipes/flux_inpaint.py
+++ b/skynet/dgpu/pipes/flux_inpaint.py
@@ -45,6 +45,7 @@ def pipeline_for(
         torch_dtype=torch.bfloat16,
         quantization_config=qonfig
     )
+    params['transformer'] = transformer
 
     pipe = FluxFillPipeline.from_pretrained(
         model, **params)