Fix ui exception