Fix main layout