fix break and return statements inside for-in and for-of