lib/berkshelf/lockfile.rb in berkshelf-0.4.0.rc4 vs lib/berkshelf/lockfile.rb in berkshelf-0.4.0

@@ -26,12 +26,12 @@
     private
 
       def get_source_definition(source)
         definition = "cookbook '#{source.name}'"
 
-        if source.location.is_a?(CookbookSource::GitLocation)
+        if source.location.is_a?(GitLocation)
           definition += ", :git => '#{source.location.uri}', :ref => '#{source.location.branch || 'HEAD'}'"
-        elsif source.location.is_a?(CookbookSource::PathLocation)
+        elsif source.location.is_a?(PathLocation)
           definition += ", :path => '#{source.location.path}'"
         else
           definition += ", :locked_version => '#{source.locked_version}'"
         end