Fix ui display error