Fixed a little issue with the main window design