Fix edit bug :grin: