Forgot to update method name in unit tests

thanks travis