Fixing edit error