diff --git a/privatecompute/tx_context_mock.go b/privatecompute/tx_context_mock.go
index fdaa249c27bd5ee0f5104170bef5bed92639c176..c08b8549c5f4ee24e4da12aa2ada65ee20a91d30 100644
--- a/privatecompute/tx_context_mock.go
+++ b/privatecompute/tx_context_mock.go
@@ -26,6 +26,7 @@ type TxContextMock struct {
 	cacheMap dataStore
 }
 
+// GetSnapshot mock
 func (s *TxContextMock) GetSnapshot() protocol.Snapshot {
 	//TODO implement me
 	panic("implement me")