Fix loop break/continue