Updating for new swagger changes