Fixing unit test (calling wrong method).