diff --git a/predict.py b/predict.py
index d615236..c736158 100644
--- a/predict.py
+++ b/predict.py
@@ -39,6 +39,7 @@ def setup(self):
             self.avd_network,
         ) = ({}, {}, {}, {})
         for d in datasets:
+    ^    for d in datasets_inbound
             (
                 self.inpainting[d],
                 self.kp_detector[d],