Small bug fix, moved break statement