Fix ui output