diff --git a/ipamd/datastore/data_store_test.go b/ipamd/datastore/data_store_test.go
index 7c721f8f51..b6016ebbc0 100644
--- a/ipamd/datastore/data_store_test.go
+++ b/ipamd/datastore/data_store_test.go
@@ -67,7 +67,7 @@ func TestDeleteENI(t *testing.T) {
 	assert.Equal(t, len(eniInfos.ENIIPPools), 2)
 
 	// Add an IP and assign a pod.
-	err = ds.AddIPv4AddressFromStore("eni-1", "1.1.1.1")
+	err = ds.AddIPv4AddressToStore("eni-1", "1.1.1.1")
 	assert.NoError(t, err)
 	podInfo := &k8sapi.K8SPodInfo{
 		Name:      "pod-1",