fixed bug in for/do/while/switch