Update unit test method signature