diff --git a/server/site_tests/firmware_ECWriteProtect/control.dev b/server/site_tests/firmware_ECWriteProtect/control.dev
index fb53054..9d66f8c 100644
--- a/server/site_tests/firmware_ECWriteProtect/control.dev
+++ b/server/site_tests/firmware_ECWriteProtect/control.dev
@@ -31,7 +31,7 @@
 """
 
 args_dict = utils.args_to_dict(args)
-servo_args = hosts.SiteHost.get_servo_arguments(args_dict)
+servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
 
 def run_ecwriteprotect(machine):
     host = hosts.create_host(machine, servo_args=servo_args)