diff --git a/pkg/builder_test.go b/pkg/builder_test.go
index 36143af..cc2dc3f 100644
--- a/pkg/builder_test.go
+++ b/pkg/builder_test.go
@@ -19,9 +19,7 @@ package statediff_test
 import (
 	"bytes"
 	"encoding/json"
-	"fmt"
 	"math/big"
-	"os"
 	"sort"
 	"testing"
 
@@ -498,13 +496,6 @@ var (
 	})
 )
 
-func init() {
-	if os.Getenv("MODE") != "statediff" {
-		fmt.Println("Skipping statediff test")
-		os.Exit(0)
-	}
-}
-
 func TestBuilder(t *testing.T) {
 	blocks, chain := test_helpers.MakeChain(3, test_helpers.Genesis, test_helpers.TestChainGen)
 	contractLeafKey = test_helpers.AddressToLeafKey(test_helpers.ContractAddr)