lib/generators/automation/templates/account.tt in ruby_raider-0.9.3 vs lib/generators/automation/templates/account.tt in ruby_raider-0.9.4

@@ -1,5 +1,5 @@
-<%- if automation == 'selenium' -%>
+<%- if selenium_based? -%>
 <%=- ERB.new(File.read(File.expand_path('./partials/selenium_account.tt', __dir__)), trim_mode: '-').result(binding) -%>
-<%- elsif automation == 'watir' -%>
+<%- elsif watir? -%>
 <%=- ERB.new(File.read(File.expand_path('./partials/watir_account.tt', __dir__)), trim_mode: '-').result(binding) -%>
 <%- end -%>
\ No newline at end of file