UI error fixed