Handle block comments correctly