diff --git a/image_prediction/predictor/__init__.py b/image_prediction/predictor/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/test/unit_tests/conftest.py b/test/unit_tests/conftest.py
index 5de1f69..7886cb9 100644
--- a/test/unit_tests/conftest.py
+++ b/test/unit_tests/conftest.py
@@ -22,13 +22,6 @@ def estimator(estimator_adapter, classes):
     return estimator
 
 
-@pytest.fixture
-def input_output_mapper(input_batch, classes):
-    """Mocks the internal, real estimator of an EstimatorAdapter object."""
-    outputs = random.choices(range(len(classes)), k=len(input_batch))
-    return outputs
-
-
 @pytest.fixture
 def estimator_adapter(estimator_type, keras_model, output_batch_generator, monkeypatch):
     if estimator_type == "mock":