Fixed loop annotations for while-expression