Fixes highighting of loop blocks when checking timeouts.