Fix spec for take_screenshots

elixir-wallaby · Apr 18, 2018 · d5f07ac · d5f07ac
1 parent 6243b7f
commit d5f07ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/wallaby/experimental/selenium/webdriver_client.ex b/lib/wallaby/experimental/selenium/webdriver_client.ex
@@ -233,7 +233,7 @@ defmodule Wallaby.Experimental.Selenium.WebdriverClient do
   @doc """
   Takes a screenshot.
   """
-  @spec take_screenshot(Session.t) :: binary
+  @spec take_screenshot(Session.t | Element.t) :: binary
   def take_screenshot(session) do
     with {:ok, resp}   <- request(:get, "#{session.session_url}/screenshot"),
           {:ok, value}  <- Map.fetch(resp, "value"),