diff --git a/.coveragerc b/.coveragerc
new file mode 100644
index 0000000..147c6da
--- /dev/null
+++ b/.coveragerc
@@ -0,0 +1,10 @@
+[run]
+source = 
+    ChemSpaceAL/InitializeWorkspace.py
+
+omit =
+    ChemSpaceAL/Model.py
+    ChemSpaceAL/Dataset.py
+    ChemSpaceAL/Docking.py
+    ChemSpaceAL/Generation.py
+    ChemSpaceAL/Training.py
diff --git a/ChemSpaceAL/Configuration.py b/ChemSpaceAL/Configuration.py
index 48d1943..69a8399 100644
--- a/ChemSpaceAL/Configuration.py
+++ b/ChemSpaceAL/Configuration.py
@@ -2,7 +2,7 @@
 from torch.nn import Module
 import os
 from typing import Union, Dict, Any, List, Tuple, Optional, Any, Set, Callable, cast
-from ChemSpaceAL.InitializeWorkspace import FOLDER_STRUCTURE as fldr_struc
+from .InitializeWorkspace import FOLDER_STRUCTURE as fldr_struc
 import textwrap
 from rdkit.Chem import Descriptors
 from rdkit import Chem