lib/appium_lib/driver.rb in appium_lib-9.7.3 vs lib/appium_lib/driver.rb in appium_lib-9.7.4

@@ -146,10 +146,11 @@
       @appium_debug = appium_lib_opts.fetch :debug, !!defined?(Pry)
       set_sauce_related_values(appium_lib_opts)
 
       # Extend Common methods
       extend Appium::Common
+      extend Appium::Device
 
       # Extend each driver's methods
       extend_for(device: @core.device, automation_name: @core.automation_name)
 
       # for command
@@ -382,10 +383,11 @@
     #   screenshot '/tmp/hi.png'
     #
     # @param png_save_path [String] the full path to save the png
     # @return [nil]
     def screenshot(png_save_path)
-      @core.screenshot png_save_path
+      @driver.save_screenshot png_save_path
+      nil
     end
 
     # Quits the driver
     # @return [void]
     def driver_quit