trying to fix the UI alignment