fixing edit screen