Make method protected, fix failing test