fixed mafia functionality