fixing edit stuff