diff --git a/heimdall/take_screenshot.py b/heimdall/take_screenshot.py
index <HASH>..<HASH> 100644
--- a/heimdall/take_screenshot.py
+++ b/heimdall/take_screenshot.py
@@ -1,4 +1,4 @@
-from heimdall import screenshot
+from heimdall.heimdall import screenshot
 
 
 if __name__ == '__main__':
update take_screenshot.py

Update needed in order for heimdall to work with Python3