Fix Main screen error