diff --git a/tests/Makefile b/tests/Makefile
index 5e1458a..8b2af47 100644
--- a/tests/Makefile
+++ b/tests/Makefile
@@ -50,6 +50,15 @@
 LOCAL_LDFLAGS += -lz
 endif
 
+# Python executable
+ifndef PYTHON
+ifdef WINDOWS
+PYTHON=python
+else
+PYTHON=python2
+endif
+endif
+
 APPS =	test_suite_aes.ecb	test_suite_aes.cbc		\
 	test_suite_aes.cfb	test_suite_aes.ofb		\
 	test_suite_aes.xts					\
@@ -188,7 +197,7 @@
 .SECONDEXPANSION:
 $(C_FILES): %.c: suites/$$(func.$$*).function suites/%.data scripts/generate_code.py suites/helpers.function suites/main_test.function suites/host_test.function
 	echo "  Gen   $@"
-	python scripts/generate_code.py -f suites/$(func.$*).function \
+	$(PYTHON) scripts/generate_code.py -f suites/$(func.$*).function \
 		-d suites/$*.data \
 		-t suites/main_test.function \
 		-p suites/host_test.function \
@@ -223,7 +232,7 @@
 .SECONDEXPANSION:
 $(EMBEDDED_TESTS): embedded_%: suites/$$(func.$$*).function suites/%.data scripts/generate_code.py suites/helpers.function suites/main_test.function suites/target_test.function
 	echo "  Gen  ./TESTS/mbedtls/$*/$*.c"
-	python scripts/generate_code.py -f suites/$(func.$*).function \
+	$(PYTHON) scripts/generate_code.py -f suites/$(func.$*).function \
 		-d suites/$*.data \
 		-t suites/main_test.function \
 		-p suites/target_test.function \