lib/mongoid/shell/properties/host.rb in mongoid-shell-0.2.0 vs lib/mongoid/shell/properties/host.rb in mongoid-shell-0.3.0

@@ -1,20 +1,22 @@
 module Mongoid
   module Shell
     module Properties
       module Host
-
         attr_accessor :host
 
         # database host
         def host
           @host || begin
             node = session.cluster.nodes.first
             raise Mongoid::Shell::Errors::SessionNotConnectedError unless node
-            node.address == "localhost:27017" ? nil : node.address
+            if Mongoid::Shell.mongoid3?
+              node.address == "localhost:27017" ? nil : node.address
+            else
+              node.address.original == "localhost:27017" ? nil : node.address.original
+            end
           end
         end
-
       end
     end
   end
 end