From 218da357c1add9bf5cbba890446838d215996aa5 Mon Sep 17 00:00:00 2001
From: Tim Band <t.b@ucl>
Date: Mon, 29 Jun 2026 15:37:09 +0100
Subject: [PATCH] remove-vocab no longer breaks DuckDB tests

---
 datafaker/create.py      | 1 +
 tests/test_functional.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/datafaker/create.py b/datafaker/create.py
index d58b7af..849a1ec 100644
--- a/datafaker/create.py
+++ b/datafaker/create.py
@@ -136,6 +136,7 @@ def create_db_vocab(
         config,
         dst_engine,
     )
+    dst_engine.dispose()
     return tables_loaded
 
 
diff --git a/tests/test_functional.py b/tests/test_functional.py
index f12786b..ca5680a 100644
--- a/tests/test_functional.py
+++ b/tests/test_functional.py
@@ -317,7 +317,7 @@ class DBFunctionalTestCase(DBFunctionalTestCaseBase):
     schema_name = "public"
     alt_orm_file_path = Path("my_orm.yaml")
 
-    def atest_workflow_maximal_args(self) -> None:
+    def test_workflow_maximal_args(self) -> None:
         """Test the CLI workflow runs with optional arguments."""
         completed_process = self.invoke(
             "--verbose",