1 files changed, 6 insertions, 0 deletions

diff --git a/plugin_python.c b/plugin_python.c
index dada5dc..0f52bb3 100644
--- a/plugin_python.c
+++ b/plugin_python.c
@@ -2,7 +2,13 @@
 #include <stdio.h>
 #include "trace-cmd.h"
 
+#ifndef PYTHON_DIR
+#define PYTHON_DIR .
+#endif
+
 static const char pyload[] =
+"import sys\n"
+"sys.path.append(\"" MAKE_STR(PYTHON_DIR) "\")\n"
 "import imp, tracecmd, ctracecmd\n"
 "fn = r'%s'\n"
 "file = open(fn, 'r')\n"