Fix reference to removed method