diff --git a/make_prob.py b/make_prob.py
index d05c633..1956831 100755
--- a/make_prob.py
+++ b/make_prob.py
@@ -5,8 +5,8 @@ import subprocess
 import shutil
 
 # set up this in order to point to proper directory.
-setup_dir = "/home/junikim/cp"
-os.environ["CP_SETUP_DIR"] = setup_dir
+setup_dir = os.path.dirname(__file__)
+os.environ["CP_SETUP_DIR"] = os.path.dirname(__file__)
 
 parser = argparse.ArgumentParser()