diff --git a/internal/lsp/source/format.go b/internal/lsp/source/format.go
index 35b7981..3a846bf 100644
--- a/internal/lsp/source/format.go
+++ b/internal/lsp/source/format.go
@@ -132,13 +132,13 @@
 	if err != nil {
 		return nil, nil, err
 	}
-	origImports, origImportOffset := trimToImports(view.Session().Cache().FileSet(), origAST, origData)
 
 	allFixes, err := imports.FixImports(filename, origData, options)
 	if err != nil {
 		return nil, nil, err
 	}
 
+	origImports, origImportOffset := trimToImports(view.Session().Cache().FileSet(), origAST, origData)
 	allFixEdits, err = computeFixEdits(view, ph, options, origData, origAST, origMapper, origImports, origImportOffset, allFixes)
 	if err != nil {
 		return nil, nil, err