diff --git a/examples/tests/cases/case_selenium.yaml b/examples/tests/cases/case_selenium.yaml
index 33fb2301..ae59bdec 100644
--- a/examples/tests/cases/case_selenium.yaml
+++ b/examples/tests/cases/case_selenium.yaml
@@ -30,3 +30,12 @@ TestSwitchFrameId:
       Id: "intercom-frame"
     - Type: switch_frame
       Value: "intercom-frame"
+
+TestGetSource:
+  Roles:
+    - Role: desktopChrome
+      App: desktop
+  Actions:
+    - Type: navigate
+      Value: https://www.testdevlab.com/
+    - Type: get_source
diff --git a/lib/core/device.rb b/lib/core/device.rb
index 5b6c5f55..e54d4752 100644
--- a/lib/core/device.rb
+++ b/lib/core/device.rb
@@ -835,7 +835,11 @@ def get_contexts(action)
 
   # parses and saves the source code for currect page.
   def get_source(action)
-    source = @driver.get_source
+    source = if @platform.nil? || @platform == "desktop"
+      @driver.page_source
+    else
+      @driver.get_source
+    end
     File.write("./page_source.xml", source)
   end