edit screen buttons fix