diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
index 736794a..96d702b 100644
--- a/.github/workflows/tests.yaml
+++ b/.github/workflows/tests.yaml
@@ -1,6 +1,10 @@
 name: Unittests
 
-on: [pull_request]
+on:
+  pull_request:
+  push:
+    branches:
+      - master
 
 jobs:
     unittests:
diff --git a/MANIFEST.in b/MANIFEST.in
new file mode 100644
index 0000000..a5782a5
--- /dev/null
+++ b/MANIFEST.in
@@ -0,0 +1,6 @@
+include README.md
+include LICENSE
+
+recursive-exclude test *
+recursive-include cpu *
+recursive-include cuda *
\ No newline at end of file
diff --git a/setup.py b/setup.py
index a6a4ab6..23a1420 100644
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,7 @@
 requirements = ["torch>=1.1.0"]
 
 url = 'https://github.com/nicolas-chaulet/torch-points-kernels'
-__version__="0.5.1"
+__version__="0.5.2"
 setup(
     name="torch-points-kernels",
     version=__version__,