lib/kuber_kit/ui/simple.rb in kuber_kit-0.3.5 vs lib/kuber_kit/ui/simple.rb in kuber_kit-0.3.6

@@ -69,10 +69,11 @@
   end
 
   def prompt(text, options, &callback)
     print_info("Select", text)
     result = $stdin.gets.chomp
-    callback.call(result)
+    callback.call(result) if callback
+    result
   end
 
   private
     def print_text(title, text, color:)
       puts "#{title.colorize(color)}\r\n #{text.colorize(color)}"
\ No newline at end of file