diff --git a/astrobee/survey_manager/survey_planner/CMakeLists.txt b/astrobee/survey_manager/survey_planner/CMakeLists.txt
index 9ce7f718..bae60f96 100644
--- a/astrobee/survey_manager/survey_planner/CMakeLists.txt
+++ b/astrobee/survey_manager/survey_planner/CMakeLists.txt
@@ -75,6 +75,7 @@ catkin_install_python(PROGRAMS
   tools/command_astrobee
   tools/monitor_astrobee
   tools/plan_interpreter
+  tools/plan_survey
   tools/problem_generator
   DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION})
 
diff --git a/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py b/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py
index 145662e8..2f98e080 100755
--- a/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py
+++ b/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py
@@ -29,7 +29,7 @@
 
 import pyparsing as pp
 import yaml
-from problem_generator import PDDL_DIR
+from survey_planner.problem_generator import PDDL_DIR
 
 LocationName = str  # Names PDDL object of type location
 LocationIndex = int  # Index of location in CONFIG.locations
diff --git a/astrobee/survey_manager/survey_planner/tools/plan_survey b/astrobee/survey_manager/survey_planner/tools/plan_survey
index 1a752a32..4fe23f84 100755
--- a/astrobee/survey_manager/survey_planner/tools/plan_survey
+++ b/astrobee/survey_manager/survey_planner/tools/plan_survey
@@ -20,7 +20,7 @@
 
 import sys
 
-from survey_planner import survey_planner
+from survey_planner import plan_survey
 
 if __name__ == "__main__":
-    sys.exit(survey_planner.main())
+    sys.exit(plan_survey.main())