diff --git a/drift_dev/lib/src/writer/tables/update_companion_writer.dart b/drift_dev/lib/src/writer/tables/update_companion_writer.dart
index c34f3cab..0c2f343c 100644
--- a/drift_dev/lib/src/writer/tables/update_companion_writer.dart
+++ b/drift_dev/lib/src/writer/tables/update_companion_writer.dart
@@ -271,7 +271,7 @@ class UpdateCompanionWriter {
     final info = table.existingRowClass;
     if (info == null) return;
 
-    _buffer.write('mixin ${table.nameOfRowClass}Columns ');
+    _buffer.write('mixin ${table.baseDartName}ToColumns ');
 
     final type = _emitter.dartCode(_emitter.writer.rowType(table));
     _buffer.writeln('implements ${_emitter.drift('Insertable')}<$type> {');