diff --git a/pkg/extension/server.go b/pkg/extension/server.go
index af1f8371c846550899bdf5e99e4d85021cc2073a..9e38a22f2aec12bf930160ad004952195c3581d4 100644
--- a/pkg/extension/server.go
+++ b/pkg/extension/server.go
@@ -87,9 +87,7 @@ func (srv *Server) Action(ctx context.Context, in *pb.ActionRequest) (*pb.Action
 		if err != nil {
 			return nil, err
 		}
-		if actionURL.Scheme == "grpc" {
-			ext = actionURL.Extension()
-		}
+		ext = actionURL.Extension()
 	}
 
 	svc, ok := srv.services[ext]
diff --git a/pkg/extension/service/extension.go b/pkg/extension/service/extension.go
index 47f076235b4a4a92ed55d9fdb00ebbcbd95e8eb4..215aca85ff935ba03248664abbb716d47f26b2cf 100644
--- a/pkg/extension/service/extension.go
+++ b/pkg/extension/service/extension.go
@@ -162,9 +162,7 @@ func (s *Extension) Action(ctx context.Context, in *extension.ActionRequest) (*e
 		if err != nil {
 			return nil, err
 		}
-		if actionURL.Scheme == "grpc" {
-			ext = actionURL.Extension()
-		}
+		ext = actionURL.Extension()
 	}
 	ok, err := extension.CheckInstalled(ctx, s.Content, in.SpaceId, in.EnvId, ext)
 	if err != nil {