Fixing minor ui issue