diff --git a/setup.py b/setup.py
index 2c7f16c..1c7283e 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 import os
 
-VERSION = "0.0.0a0"
+VERSION = "0.0.1a0"
 
 
 def get_long_description():
diff --git a/tests/conftest.py b/tests/conftest.py
index 8f3cb83..fb7bc56 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -32,7 +32,7 @@ def existing_db_path(fresh_db_path):
     db = Database(fresh_db_path)
     table = db.create_table(
         "prompts",
-        {"prompt": str},
+        {"prompt": str, "generated": str}
     )
 
     table.insert({"prompt": "hello world!"})
diff --git a/tests/test_tsellm.py b/tests/test_tsellm.py
index e2317ba..79df5f1 100644
--- a/tests/test_tsellm.py
+++ b/tests/test_tsellm.py
@@ -22,4 +22,3 @@ def test_cli_prompt_mock(existing_db_path):
         prompt_words = prompt.split()
         for word in words:
             assert word in prompt_words
-    assert existing_db_path == 'gdfg'
\ No newline at end of file