diff --git a/api-tests/management/annotation_test.go b/api-tests/management/annotation_test.go
index 10709f84d1..e35272cb74 100644
--- a/api-tests/management/annotation_test.go
+++ b/api-tests/management/annotation_test.go
@@ -91,7 +91,7 @@ func TestAddAnnotation(t *testing.T) {
 		resNode, err := inventoryClient.Default.Nodes.AddGenericNode(paramsNode)
 		assert.NoError(t, err)
 		genericNodeID := resNode.Payload.Generic.NodeID
-		defer pmmapitests.UnregisterNodes(t, genericNodeID)
+		defer pmmapitests.RemoveNodes(t, genericNodeID)
 
 		serviceName := pmmapitests.TestString(t, "annotation-service")
 		paramsService := &services.AddMySQLServiceParams{
@@ -131,7 +131,7 @@ func TestAddAnnotation(t *testing.T) {
 		}
 		res, err := inventoryClient.Default.Nodes.AddGenericNode(params)
 		assert.NoError(t, err)
-		defer pmmapitests.UnregisterNodes(t, res.Payload.Generic.NodeID)
+		defer pmmapitests.RemoveNodes(t, res.Payload.Generic.NodeID)
 
 		paramsAdd := &annotation.AddAnnotationParams{
 			Body: annotation.AddAnnotationBody{