2 files changed, 2 insertions, 5 deletions
diff --git a/tests/fsm/generate.py b/tests/fsm/generate.py
index fb5695ff6..8757d4741 100644
--- a/tests/fsm/generate.py
+++ b/tests/fsm/generate.py
@@ -1,7 +1,4 @@
-#!/usr/bin/python
-
-from __future__ import division
-from __future__ import print_function
+#!/usr/bin/env python3
 
 import sys
 import random
diff --git a/tests/fsm/run-test.sh b/tests/fsm/run-test.sh
index 57c2a5b12..423892334 100755
--- a/tests/fsm/run-test.sh
+++ b/tests/fsm/run-test.sh
@@ -8,7 +8,7 @@ set -e
 rm -rf temp
 mkdir -p temp
 echo "generating tests.."
-python generate.py
+python3 generate.py
 
 {
 	all_targets="all_targets:"