Update tests for new method name