diff --git a/.gitignore b/.gitignore
index 0c2c0c79e46..7664e01f5ab 100644
--- a/.gitignore
+++ b/.gitignore
@@ -14,5 +14,4 @@ venv/*
 *.key
 .eggs
 eggs/*
-*.pyi
-data/*
\ No newline at end of file
+*.pyi
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 1b843ee5998..947285c1088 100644
--- a/setup.py
+++ b/setup.py
@@ -141,7 +141,7 @@ def run(self):
         'openfl.utilities.optimizers.torch',
         'openfl-docker',
         'openfl-tutorials',
-        'openfl-workspace'
+        'openfl-workspace',
     ],
     include_package_data=True,
     install_requires=[