Fix edit header screen