1 files changed, 1 insertions, 7 deletions
diff --git a/lib/api/entities/ml/mlflow/update_run.rb b/lib/api/entities/ml/mlflow/update_run.rb
index 090d69b8895..55def810ef5 100644
--- a/lib/api/entities/ml/mlflow/update_run.rb
+++ b/lib/api/entities/ml/mlflow/update_run.rb
@@ -5,13 +5,7 @@ module API
     module Ml
       module Mlflow
         class UpdateRun < Grape::Entity
-          expose :run_info
-
-          private
-
-          def run_info
-            RunInfo.represent object
-          end
+          expose :itself, using: RunInfo, as: :run_info
         end
       end
     end