diff --git a/retrievalmarket/types.go b/retrievalmarket/types.go
index 1c56217f..fa7108d4 100644
--- a/retrievalmarket/types.go
+++ b/retrievalmarket/types.go
@@ -27,9 +27,7 @@ import (
 
 var log = logging.Logger("retrievalmrkt")
 
-//go:generate cbor-gen-for DealStage DealStages
-
-//go:generate cbor-gen-for --map-encoding Query QueryResponse DealProposal DealResponse Params QueryParams DealPayment ClientDealState ProviderDealState PaymentInfo RetrievalPeer Ask
+//go:generate cbor-gen-for --map-encoding Query QueryResponse DealProposal DealResponse Params QueryParams DealPayment ClientDealState ProviderDealState PaymentInfo RetrievalPeer Ask DealStage DealStages
 
 // QueryProtocolID is the protocol for querying information about retrieval
 // deal parameters