diff --git a/nodebuilder/pruner/module.go b/nodebuilder/pruner/module.go
index 9adfe2c4d8..f7af500a3c 100644
--- a/nodebuilder/pruner/module.go
+++ b/nodebuilder/pruner/module.go
@@ -75,8 +75,9 @@ func ConstructModule(tp node.Type, cfg *Config) fx.Option {
 			}),
 		)
 	case node.Light:
-		fx.Provide(light.NewPruner)
 		return fx.Module("prune",
+			baseComponents,
+			fx.Provide(light.NewPruner),
 			fx.Supply(light.Window),
 		)
 	default:
diff --git a/pruner/light/pruner.go b/pruner/light/pruner.go
index 6fe17040f3..e7589cebe0 100644
--- a/pruner/light/pruner.go
+++ b/pruner/light/pruner.go
@@ -8,6 +8,7 @@ import (
 	"github.com/ipfs/go-datastore"
 
 	"github.com/celestiaorg/celestia-node/header"
+	"github.com/celestiaorg/celestia-node/pruner"
 	"github.com/celestiaorg/celestia-node/share"
 	"github.com/celestiaorg/celestia-node/share/ipld"
 )
@@ -17,7 +18,7 @@ type Pruner struct {
 	ds    datastore.Datastore
 }
 
-func NewPruner(bstore blockstore.Blockstore, ds datastore.Datastore) *Pruner {
+func NewPruner(bstore blockstore.Blockstore, ds datastore.Batching) pruner.Pruner {
 	return &Pruner{bserv: blockservice.New(bstore, nil), ds: ds}
 }