Fixes edit item bug