diff --git a/chain/wallet/multi_test.go b/chain/wallet/multi_test.go
index b74f435e1..54ff240c5 100644
--- a/chain/wallet/multi_test.go
+++ b/chain/wallet/multi_test.go
@@ -1,3 +1,4 @@
+//stm: #unit
 package wallet
 
 import (
diff --git a/chain/wallet/wallet_test.go b/chain/wallet/wallet_test.go
index 4aba28707..f07a6278c 100644
--- a/chain/wallet/wallet_test.go
+++ b/chain/wallet/wallet_test.go
@@ -1,3 +1,4 @@
+//stm: #unit
 package wallet
 
 import (