diff --git a/oletools/oledir.py b/oletools/oledir.py
index 5fef39b..02b9c07 100755
--- a/oletools/oledir.py
+++ b/oletools/oledir.py
@@ -113,7 +113,7 @@ def sid_display(sid):
 
 # === MAIN ===================================================================
 
-if __name__ == '__main__':
+def main():
     # print banner with version
     print('oledir %s - http://decalage.info/python/oletools' % __version__)
 
@@ -182,3 +182,5 @@ if __name__ == '__main__':
     # print t
 
 
+if __name__ == '__main__':
+    main()
\ No newline at end of file
diff --git a/setup.py b/setup.py
index bb247d8..beb362c 100755
--- a/setup.py
+++ b/setup.py
@@ -268,6 +268,7 @@ entry_points = {
         'mraptor=oletools.mraptor:main',
         'olebrowse=oletools.olebrowse:main',
         'oleid=oletools.oleid:main',
+        'oledir=oletools.oledir:main',
         'pyxswf=oletools.pyxswf:main'
     ],
 }