lib/rash/shell/pry/commands/no_prompt.rb in rash-shell-0.1.1 vs lib/rash/shell/pry/commands/no_prompt.rb in rash-shell-0.2.0

@@ -1,12 +1,14 @@
+# frozen_string_literal: true
+
 Pry::Commands.create_command 'no-prompt' do
   description 'Show/hide REPL command prompt prefix'
 
   def process
     case _pry_.prompt
-    when Pry::NO_PROMPT
+    when Pry::Prompt[:none][:value]
       _pry_.pop_prompt
     else
-      _pry_.push_prompt Pry::NO_PROMPT
+      _pry_.push_prompt Pry::Prompt[:none][:value]
     end
   end
 end