Fix method signature in test