lib/rvmpow.rb in rvmpow-0.0.2 vs lib/rvmpow.rb in rvmpow-0.0.3

@@ -139,12 +139,20 @@
 
 			# returns a hash with the ruby and gemset information
 			# @return [Hash]
 			def rvmInfo
 				rvm = {}
-				rvmArray = `rvm-prompt`.chomp.split('@')
-				rvm[:ruby] = rvmArray[0].split('-')[1]
-				rvm[:gemset] = rvmArray[1]
+
+				rg = `rvm-prompt`
+				if !rg.include?('@')
+					rvm[:ruby] = rg.chomp
+					rvm[:gemset] = ''
+				else
+					rvmArray = rg.chomp.split('@')
+					rvm[:ruby] = rvmArray[0].split('-')[1]
+					rvm[:gemset] = rvmArray[1]
+				end
+
 				rvm
 			end
 
 			def createTmpDirIfNeeded
 				FileUtils.mkdir('tmp') if !File.exist?('tmp')
\ No newline at end of file