Fixed a string comparison in the Main classes