diff --git a/sample_pyhidra.py b/sample_pyhidra.py
index 9d61cb6..e751a1e 100644
--- a/sample_pyhidra.py
+++ b/sample_pyhidra.py
@@ -2,6 +2,7 @@
 
 # Section to make autocomplete work
 try:
+    import os
     import ghidra
     from ghidra_builtins import *
 except:
@@ -11,6 +12,8 @@
 PROJECT_NAME = 'sample_pyghidra'
 PROJECT_LOCATION = '.ghidra_projects'
 
+os.environ["JAVA_TOOL_OPTIONS"] = "-Djava.awt.headless=true"
+
 pyghidra.start(True)  # setting Verbose output
 
 from ghidra.program.model.listing import Program  # noqa