Fixed if bugs