Removed reference to removed class to fix build