'cheetah' command rewrite

2 files changed, 5 insertions, 3 deletions

diff --git a/bin/cheetah b/bin/cheetah
index 414aad6..26c391a 100755
--- a/bin/cheetah
+++ b/bin/cheetah
@@ -1,3 +1,3 @@
 #!/usr/bin/env python
 from Cheetah.CheetahWrapper import CheetahWrapper
-CheetahWrapper().run()
\ No newline at end of file
+CheetahWrapper().main()
diff --git a/bin/cheetah-compile b/bin/cheetah-compile
index f558ea2..5d9ab60 100644
--- a/bin/cheetah-compile
+++ b/bin/cheetah-compile
@@ -1,3 +1,5 @@
 #!/usr/bin/env python
-from Cheetah.CheetahCompile import CheetahCompile
-CheetahCompile().run()
\ No newline at end of file
+import sys
+from Cheetah.CheetahWrapper import CheetahWrapper
+sys.argv.insert(1, "compile")
+CheetahWrapper().main()