Fix mafia errors too