lib/appium_lib/driver.rb in appium_lib-0.18.1 vs lib/appium_lib/driver.rb in appium_lib-0.18.2

@@ -296,10 +296,10 @@
       # 'iPad Simulator'
       # 'Android'
       # 'Selendroid'
       #
       # :ios, :android, :selendroid
-      @device = opts[:device]
+      @device = opts.fetch :device, ENV['DEVICE']
       raise 'Device must be set' unless @device
 
       @device = 'Android' if @device.to_s == 'android'
       @device = 'Selendroid' if @device.to_s == 'selendroid'
       @device = 'iPhone Simulator' if @device.to_s == 'ios'
\ No newline at end of file