diff --git a/retrievalmarket/types.go b/retrievalmarket/types.go
index c46c1d0b..1c56217f 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
-
-//go:generate cbor-gen-for DealStages
+//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