diff --git a/plugins/csi/client.go b/plugins/csi/client.go
index 556f9142a..08add6719 100644
--- a/plugins/csi/client.go
+++ b/plugins/csi/client.go
@@ -67,6 +67,8 @@ type CSIControllerClient interface {
 type CSINodeClient interface {
 	NodeGetCapabilities(ctx context.Context, in *csipbv1.NodeGetCapabilitiesRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetCapabilitiesResponse, error)
 	NodeGetInfo(ctx context.Context, in *csipbv1.NodeGetInfoRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetInfoResponse, error)
+	NodeStageVolume(ctx context.Context, in *csipbv1.NodeStageVolumeRequest, opts ...grpc.CallOption) (*csipbv1.NodeStageVolumeResponse, error)
+	// NodeUnstageVolume(ctx context.Context, in *NodeUnstageVolumeRequest, opts ...grpc.CallOption) (*NodeUnstageVolumeResponse, error)
 }
 
 type client struct {
diff --git a/plugins/csi/testing/client.go b/plugins/csi/testing/client.go
index 75f20929d..f341634a8 100644
--- a/plugins/csi/testing/client.go
+++ b/plugins/csi/testing/client.go
@@ -83,6 +83,7 @@ type NodeClient struct {
 	NextErr                  error
 	NextCapabilitiesResponse *csipbv1.NodeGetCapabilitiesResponse
 	NextGetInfoResponse      *csipbv1.NodeGetInfoResponse
+	NextStageVolumeResponse  *csipbv1.NodeStageVolumeResponse
 }
 
 // NewNodeClient returns a new ControllerClient
@@ -103,3 +104,7 @@ func (c *NodeClient) NodeGetCapabilities(ctx context.Context, in *csipbv1.NodeGe
 func (c *NodeClient) NodeGetInfo(ctx context.Context, in *csipbv1.NodeGetInfoRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetInfoResponse, error) {
 	return c.NextGetInfoResponse, c.NextErr
 }
+
+func (c *NodeClient) NodeStageVolume(ctx context.Context, in *csipbv1.NodeStageVolumeRequest, opts ...grpc.CallOption) (*csipbv1.NodeStageVolumeResponse, error) {
+	return c.NextStageVolumeResponse, c.NextErr
+}