diff --git a/pkg/extension/action_test.go b/pkg/extension/action_test.go
index 3741b5af10c16c9a6d84ba890b1a7800512cde2b..edba7e4712f2f334b50a0e2aa13e8ebb79c923d6 100644
--- a/pkg/extension/action_test.go
+++ b/pkg/extension/action_test.go
@@ -11,7 +11,9 @@ import (
 func TestAction(t *testing.T) {
 	t.Run("Protobuf", func(t *testing.T) {
 		var pbAction *pb.Action
-		gofakeit.Struct(&pbAction)
+		err := gofakeit.Struct(&pbAction)
+		require.NoError(t, err)
+		
 		action := ActionFromPB(pbAction)
 		result := ActionToPB(action)
 		require.Equal(t, pbAction, result)
@@ -26,7 +28,9 @@ func TestAction(t *testing.T) {
 
 	t.Run("Map", func(t *testing.T) {
 		var action *Action
-		gofakeit.Struct(&action)
+		err := gofakeit.Struct(&action)
+		require.NoError(t, err)
+		
 		dict := ActionToMap(action)
 		result, err := ActionFromMap(dict)
 		require.NoError(t, err)
@@ -45,7 +49,9 @@ func TestAction(t *testing.T) {
 func TestActionRequest(t *testing.T) {
 	t.Run("Protobuf", func(t *testing.T) {
 		var pbAction *pb.ActionRequest
-		gofakeit.Struct(&pbAction)
+		err := gofakeit.Struct(&pbAction)
+		require.NoError(t, err)
+		
 		action := ActionRequestFromPB(pbAction)
 		result := ActionRequestToPB(action)
 		require.Equal(t, pbAction, result)
@@ -62,7 +68,9 @@ func TestActionRequest(t *testing.T) {
 func TestActionResponse(t *testing.T) {
 	t.Run("Protobuf", func(t *testing.T) {
 		var pbAction *pb.ActionResponse
-		gofakeit.Struct(&pbAction)
+		err := gofakeit.Struct(&pbAction)
+		require.NoError(t, err)
+
 		action := ActionResponseFromPB(pbAction)
 		result := ActionResponseToPB(action)
 		require.Equal(t, pbAction, result)