change 'return' to 'break'