diff --git a/pkg/controller/nodes/task/catalog/datacatalog/datacatalog_test.go b/pkg/controller/nodes/task/catalog/datacatalog/datacatalog_test.go
index 581b5ddbe..72d9a4efc 100644
--- a/pkg/controller/nodes/task/catalog/datacatalog/datacatalog_test.go
+++ b/pkg/controller/nodes/task/catalog/datacatalog/datacatalog_test.go
@@ -553,13 +553,13 @@ func TestCatalog_GetOrExtendReservation(t *testing.T) {
 
 	heartbeatInterval := time.Second * 5
 	prevReservation := datacatalog.Reservation{
-		ReservationId:     &reservationID,
-		OwnerId:           prevOwner,
+		ReservationId: &reservationID,
+		OwnerId:       prevOwner,
 	}
 
 	currentReservation := datacatalog.Reservation{
-		ReservationId:     &reservationID,
-		OwnerId:           currentOwner,
+		ReservationId: &reservationID,
+		OwnerId:       currentOwner,
 	}
 
 	t.Run("CreateOrUpdateReservation", func(t *testing.T) {
diff --git a/pkg/controller/nodes/task/handler_test.go b/pkg/controller/nodes/task/handler_test.go
index 96e19a8ae..22787dd70 100644
--- a/pkg/controller/nodes/task/handler_test.go
+++ b/pkg/controller/nodes/task/handler_test.go
@@ -23,8 +23,8 @@ import (
 	"github.com/flyteorg/flytestdlib/promutils/labeled"
 
 	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core"
-	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/event"
 	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/datacatalog"
+	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/event"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery"
 	pluginCatalogMocks "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/catalog/mocks"
 	pluginCore "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/core"
@@ -972,7 +972,7 @@ func Test_task_Handle_Reservation(t *testing.T) {
 			Id:   taskID,
 			Type: "test",
 			Metadata: &core.TaskMetadata{
-				Discoverable: true,
+				Discoverable:          true,
 				DiscoverySerializable: true,
 			},
 			Interface: &core.TypedInterface{