Fixing the UI

For it is simple