diff --git a/tests/conftest.py b/tests/conftest.py
index 7c80754..158fbf5 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,10 +1,9 @@
-from distutils.command.config import config
 import pytest
-from pyinfra.config import get_config
+from pyinfra.config import get_config, Config
 import os
 
-@pytest.fixture(params=)
-def storage_config(request):
+@pytest.fixture(params=["s3", "aks"])
+def storage_config(request) -> Config:
     if request.param == "s3":
         os.environ["STORAGE_BACKEND"] = "s3"
         os.environ["STORAGE_BUCKET_NAME"] = "pyinfra-test-bucket"