diff --git a/Makefile b/Makefile
index 32c5a63..27ffbaa 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,10 @@
+ifeq ($(shell uname), Darwin)
+	PY_EXT := dylib
+else
+	PY_EXT := so
+endif
+
 test-python:
 	cargo build --release
-	cp target/release/libinstant_segment.dylib instant-segment-py/test/instant_segment.so
+	cp target/release/libinstant_segment.$(PY_EXT) instant-segment-py/test/instant_segment.so
 	PYTHONPATH=instant-segment-py/test/ python3 -m test