fixed broken edit feature