From 2dd8d2faead46d82ef36c77122923ec45cd26439 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radu=20Lucu=C8=9B?= <git@radulucut.com>
Date: Thu, 31 Oct 2024 14:38:27 +0200
Subject: [PATCH] fix test

---
 cmd/ping_test.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cmd/ping_test.go b/cmd/ping_test.go
index 5ccaf8e..ccbf1ef 100644
--- a/cmd/ping_test.go
+++ b/cmd/ping_test.go
@@ -344,7 +344,9 @@ func Test_Execute_Ping_Infinite(t *testing.T) {
 		Limit:   1,
 	}
 	_storage := createDefaultStorage(utilsMock)
-	defer _storage.Remove()
+	t.Cleanup(func() {
+		_storage.Remove()
+	})
 	root := NewRoot(printer, ctx, viewerMock, utilsMock, gbMock, nil, _storage)
 	os.Args = []string{"globalping", "ping", "jsdelivr.com", "--infinite", "from", "Berlin"}