diff --git a/lumberjack_test.go b/lumberjack_test.go
index c11dc18..8c0e685 100644
--- a/lumberjack_test.go
+++ b/lumberjack_test.go
@@ -10,7 +10,7 @@ import (
 	"time"
 
 	"github.com/BurntSushi/toml"
-	"gopkg.in/yaml.v1"
+	"gopkg.in/yaml.v2"
 )
 
 // !!!NOTE!!!