diff --git a/drift_dev/lib/src/writer/writer.dart b/drift_dev/lib/src/writer/writer.dart
index 06e15656..01d67107 100644
--- a/drift_dev/lib/src/writer/writer.dart
+++ b/drift_dev/lib/src/writer/writer.dart
@@ -198,7 +198,9 @@ abstract class _NodeOrWriter {
     final buffer = StringBuffer();
 
     for (final lexeme in code.elements) {
-      if (lexeme is DartTopLevelSymbol) {
+      if (lexeme is AnnotatedDartCode) {
+        buffer.write(dartCode(lexeme));
+      } else if (lexeme is DartTopLevelSymbol) {
         final uri = lexeme.importUri;
 
         if (uri != null) {