From eae388a161de1a81b81bedfe69de41032ee5b772 Mon Sep 17 00:00:00 2001
From: John Mair <jrmair@gmail.com>
Date: Thu, 19 Apr 2012 11:53:53 +1200
Subject: [PATCH] make correct_indentation color respect Pry.color

---
 lib/pry/indent.rb | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/pry/indent.rb b/lib/pry/indent.rb
index eea12730..f97aa70d 100644
--- a/lib/pry/indent.rb
+++ b/lib/pry/indent.rb
@@ -17,6 +17,8 @@ class Pry
   # will be indented or un-indented by correctly.
   #
   class Indent
+    include Helpers::BaseHelpers
+
     # String containing the spaces to be inserted before the next line.
     attr_reader :indent_level
 
@@ -224,7 +226,7 @@ class Pry
       end
       whitespace = ' ' * overhang
 
-      "#{move_up}#{prompt}#{CodeRay.scan(code, :ruby).term}#{whitespace}#{move_down}"
+      "#{move_up}#{prompt}#{colorize_code(code)}#{whitespace}#{move_down}"
     end
   end
 end