Fix comments tests