Fix test so it doesn't rely on removed method