diff --git a/datalite/__init__.py b/datalite/__init__.py
index 4acb744..b01f482 100644
--- a/datalite/__init__.py
+++ b/datalite/__init__.py
@@ -1,12 +1,2 @@
 __all__ = ['commons', 'datalite_decorator', 'fetch', 'migrations', 'datalite', 'constraints']
-
-from dataclasses import dataclass
-
-
 from .datalite_decorator import datalite
-
-
-@dataclass
-class Student:
-    id_: int
-    name: str
diff --git a/datalite/datalite_decorator.py b/datalite/datalite_decorator.py
index 78a0a04..60b6ff5 100644
--- a/datalite/datalite_decorator.py
+++ b/datalite/datalite_decorator.py
@@ -3,11 +3,11 @@ Defines the Datalite decorator that can be used to convert a dataclass to
 a class bound to a sqlite3 database.
 """
 from sqlite3.dbapi2 import IntegrityError
-from typing import Dict, Optional, List, Callable
-from dataclasses import Field, asdict
+from typing import Dict, Optional, Callable
+from dataclasses import asdict
 import sqlite3 as sql
 
-from constraints import ConstraintFailedError
+from .constraints import ConstraintFailedError
 from .commons import _convert_sql_format, _convert_type, _create_table, type_table
 
 
diff --git a/setup.py b/setup.py
index 4910247..8e69f18 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="datalite", # Replace with your own username
-    version="0.5.8",
+    version="0.5.9",
     author="Ege Ozkan",
     author_email="egeemirozkan24@gmail.com",
     description="A small package that binds dataclasses to an sqlite database",