Fix edit item function