tests: refactor conftest table dropping code

1 files changed, 3 insertions, 3 deletions

diff --git a/yoyo/tests/conftest.py b/yoyo/tests/conftest.py
index 1bf8d4a..f24509d 100644
--- a/yoyo/tests/conftest.py
+++ b/yoyo/tests/conftest.py
@@ -56,9 +56,9 @@ def drop_all_tables(backend):
 
 
 def drop_yoyo_tables(backend):
-    for table in backend.list_tables():
-        if table.startswith("yoyo") or table.startswith("_yoyo"):
-            with backend.transaction():
+    with backend.transaction():
+        for table in backend.list_tables():
+            if table.startswith("yoyo") or table.startswith("_yoyo"):
                 backend.execute("DROP TABLE {}".format(table))