diff --git a/scripts/flash_klipper.sh b/scripts/flash_klipper.sh
index be92d6a..6d61b0f 100644
--- a/scripts/flash_klipper.sh
+++ b/scripts/flash_klipper.sh
@@ -272,7 +272,9 @@ function build_fw(){
     local dep=(build-essential dpkg-dev make)
     dependency_check "${dep[@]}"
 
-    make clean && make menuconfig
+    make clean
+    [ "${python_version}" == "3" ] && make PYTHON=python3 menuconfig
+    [ "${python_version}" == "2" ] && make menuconfig
 
     status_msg "Building firmware ..."
     python_version=$("${KLIPPY_ENV}"/bin/python --version 2>&1 | cut -d" " -f2 | cut -d"." -f1)