lib/snapshot/runner.rb in snapshot-1.10.0 vs lib/snapshot/runner.rb in snapshot-1.11.0

@@ -173,10 +173,13 @@
                                                   UI.crash!("Too many errors... no more retries...")
                                                 end
                                               end)
 
       raw_output = File.read(TestCommandGenerator.xcodebuild_log_path)
-      return Collector.fetch_screenshots(raw_output, language, device_type, launch_arguments.first)
+
+      dir_name = locale || language
+
+      return Collector.fetch_screenshots(raw_output, dir_name, device_type, launch_arguments.first)
     end
 
     def uninstall_app(device_type)
       UI.verbose "Uninstalling app '#{Snapshot.config[:app_identifier]}' from #{device_type}..."
       Snapshot.config[:app_identifier] ||= ask("App Identifier: ")