Fixed issue with break after if