diff --git a/build.sh b/build.sh
index 8e1ab520..53a8fcfd 100755
--- a/build.sh
+++ b/build.sh
@@ -8,10 +8,7 @@ make
 # install python package
 
 cd python-package
-if command -v python2; then
-    python2 setup.py install
+if command -v python; then
+    python setup.py install
 fi
 
-if command -v python3; then
-    python3 setup.py install
-fi