diff --git a/modules/store/src/main/scala/db/migration/MigrationTasks.scala b/modules/store/src/main/scala/db/migration/MigrationTasks.scala
index f8e2bf55..39f53f58 100644
--- a/modules/store/src/main/scala/db/migration/MigrationTasks.scala
+++ b/modules/store/src/main/scala/db/migration/MigrationTasks.scala
@@ -15,7 +15,7 @@ import docspell.common.syntax.StringSyntax._
 import docspell.notification.api._
 import docspell.store.records._
 
-import db.migration.data.{PeriodicDueItemsArgsOld, PeriodicQueryArgsOld}
+import db.migration.data._
 import doobie._
 import doobie.implicits._
 import doobie.util.transactor.Strategy
diff --git a/modules/store/src/main/scala/db/migration/data/NotifyDueItemsArgs.scala b/modules/store/src/main/scala/db/migration/data/NotifyDueItemsArgs.scala
index e664b58f..0e88df9f 100644
--- a/modules/store/src/main/scala/db/migration/data/NotifyDueItemsArgs.scala
+++ b/modules/store/src/main/scala/db/migration/data/NotifyDueItemsArgs.scala
@@ -4,8 +4,9 @@
  * SPDX-License-Identifier: AGPL-3.0-or-later
  */
 
-package docspell.common
+package db.migration.data
 
+import docspell.common._
 import docspell.common.syntax.all._
 
 import io.circe._