Updated tests

2 files changed, 7 insertions, 4 deletions

diff --git a/dagger_test.go b/dagger_test.go
index 58cdd6afd..616577a39 100644
--- a/dagger_test.go
+++ b/dagger_test.go
@@ -1,13 +1,14 @@
 package main
 
 import (
+	"github.com/ethereum/ethutil-go"
 	"math/big"
 	"testing"
 )
 
 func BenchmarkDaggerSearch(b *testing.B) {
 	hash := big.NewInt(0)
-	diff := BigPow(2, 36)
+	diff := ethutil.BigPow(2, 36)
 	o := big.NewInt(0) // nonce doesn't matter. We're only testing against speed, not validity
 
 	// Reset timer so the big generation isn't included in the benchmark
diff --git a/test_runner_test.go b/test_runner_test.go
index 5abe20002..5abf0c8df 100644
--- a/test_runner_test.go
+++ b/test_runner_test.go
@@ -3,6 +3,8 @@ package main
 import (
 	"encoding/hex"
 	_ "fmt"
+	"github.com/ethereum/ethdb-go"
+	"github.com/ethereum/ethutil-go"
 	"testing"
 )
 
@@ -15,8 +17,8 @@ var testsource = `{"Inputs":{
 }`
 
 func TestTestRunner(t *testing.T) {
-	db, _ := NewMemDatabase()
-	trie := NewTrie(db, "")
+	db, _ := ethdb.NewMemDatabase()
+	trie := ethutil.NewTrie(db, "")
 
 	runner := NewTestRunner(t)
 	runner.RunFromString(testsource, func(source *TestSource) {
@@ -24,7 +26,7 @@ func TestTestRunner(t *testing.T) {
 			trie.Update(key, value)
 		}
 
-		if hex.EncodeToString([]byte(trie.root)) != source.Expectation {
+		if hex.EncodeToString([]byte(trie.Root)) != source.Expectation {
 			t.Error("trie root did not match")
 		}
 	})