diff --git a/lib/wallet/test/unit/Cardano/Wallet/Address/Discovery/SharedSpec.hs b/lib/wallet/test/unit/Cardano/Wallet/Address/Discovery/SharedSpec.hs
index f298dc9671d..677a00f72bd 100644
--- a/lib/wallet/test/unit/Cardano/Wallet/Address/Discovery/SharedSpec.hs
+++ b/lib/wallet/test/unit/Cardano/Wallet/Address/Discovery/SharedSpec.hs
@@ -53,6 +53,7 @@ import Cardano.Wallet.Address.Discovery.Shared
 import Cardano.Wallet.Address.Keys.Shared
     ( mkSharedStateFromAccountXPub, mkSharedStateFromRootXPrv )
 import Cardano.Wallet.Flavor
+    ( KeyFlavorS (SharedKeyS) )
 import Cardano.Wallet.Gen
     ( genNatural, genScript, genScriptTemplate )
 import Cardano.Wallet.Primitive.Passphrase.Gen