Fix tests for new method.