diff --git a/src/Yam/DataSource.hs b/src/Yam/DataSource.hs
--- a/src/Yam/DataSource.hs
+++ b/src/Yam/DataSource.hs
@@ -50,10 +50,10 @@
 selectValue :: (PersistField a, MonadUnliftIO m) => Text -> DB m [a]
 selectValue sql = fmap unSingle <$> rawSql sql []
 
-runTransWith :: (MonadUnliftIO m) => Key DataSource -> DB (AppM m) a -> AppM m a
+runTransWith :: Key DataSource -> DB App a -> App a
 runTransWith k a = requireAttr k >>= (`runDB` a)
 
-runTrans :: (MonadUnliftIO m) => DB (AppM m) a -> AppM m a
+runTrans :: DB App a -> App a
 runTrans = runTransWith dataSourceKey
 
 {-# INLINE runDB #-}
diff --git a/yam-datasource.cabal b/yam-datasource.cabal
--- a/yam-datasource.cabal
+++ b/yam-datasource.cabal
@@ -1,6 +1,6 @@
 cabal-version: 1.12
 name: yam-datasource
-version: 0.5.3
+version: 0.5.4
 license: BSD3
 license-file: LICENSE
 copyright: (c) Daniel YU
@@ -42,4 +42,4 @@
         resource-pool >=0.2.3.2 && <0.3,
         resourcet >=1.2.2 && <1.3,
         unliftio-core >=0.1.2.0 && <0.2,
-        yam >=0.5.3 && <0.6
+        yam >=0.5.4 && <0.6