Fix Swagger models integration