diff --git a/Makefile b/Makefile
index f2dc407..6f7cef9 100644
--- a/Makefile
+++ b/Makefile
@@ -36,7 +36,7 @@ coveralls: venv
 	$(VENV_RUN); coveralls
 
 dist: venv
-	$(VENV_RUN); python setup.py sdist bdist_wheel
+	$(VENV_RUN); pip install --upgrade build; python -m build
 
 deploy: clean-dist venv test dist
 	$(VENV_RUN); pip install --upgrade twine; twine upload dist/*
diff --git a/localstack_snapshot/pytest/snapshot.py b/localstack_snapshot/pytest/snapshot.py
index 5d1cd62..c6885d1 100644
--- a/localstack_snapshot/pytest/snapshot.py
+++ b/localstack_snapshot/pytest/snapshot.py
@@ -15,6 +15,7 @@
 from localstack_snapshot.snapshots.report import render_report
 
 
+# TODO: move?
 def is_aws():
     return os.environ.get("TEST_TARGET", "") == "AWS_CLOUD"