diff --git a/zk/conn_test.go b/zk/conn_test.go
index ed4a7706..32e78503 100644
--- a/zk/conn_test.go
+++ b/zk/conn_test.go
@@ -30,7 +30,7 @@ func TestRecurringReAuthHang(t *testing.T) {
 	if err != nil {
 		panic(err)
 	}
-	for conn.state != StateHasSession {
+	for conn.State() != StateHasSession {
 		time.Sleep(50 * time.Millisecond)
 	}
 
@@ -40,7 +40,7 @@ func TestRecurringReAuthHang(t *testing.T) {
 	}()
 
 	// Add auth.
-	conn.creds = append(conn.creds, authCreds{"digest", []byte("test:test")})
+	conn.AddAuth("digest", []byte("test:test"))
 
 	currentServer := conn.server
 	conn.debugCloseRecvLoop = true