lib/ronin/exploits/cli/commands/run.rb in ronin-exploits-1.1.0.rc1 vs lib/ronin/exploits/cli/commands/run.rb in ronin-exploits-1.1.0.rc2

@@ -281,19 +281,19 @@
 
             if options[:test]
               run_test
             else
               run_exploit
-            end
 
-            if options[:irb]
-              start_shell
-            else
-              post_exploitation
-            end
+              if options[:irb]
+                start_shell
+              else
+                post_exploitation
+              end
 
-            perform_cleanup
+              perform_cleanup
+            end
           end
 
           #
           # Loads the payload encoder classes specified by `--encoder` or
           # `--encoder-file`.
@@ -427,10 +427,10 @@
                @exploit.payload.kind_of?(Ronin::Payloads::Mixins::PostEx)
               unless @exploit.payload.session
                 print_error "payload (#{@exploit.payload.class_id}) did not create a post-exploitation session"
 
                 perform_cleanup
-                eixt(1)
+                exit(1)
               end
 
               @exploit.payload.session.system.interact
             elsif @exploit_class.include?(Mixins::Loot)
               print_loot