lib/headdesk/apk.rb in headdesk-0.11.2 vs lib/headdesk/apk.rb in headdesk-0.11.3

@@ -66,11 +66,11 @@
     def class?(decl)
       !find_class(decl).nil?
     end
 
     def find_class(decl)
-      file_name = File.join(@path, 'smali', "#{Class.path_for(decl)}.smali")
-      return nil unless File.exist? file_name
+      file_name = Dir["#{@path}/smali*/**/#{Class.path_for(decl)}.smali"].first
+      return nil unless file_name && File.exist?(file_name)
 
       Class.new(file_name)
     end
   end
 end