Fixed small bug in modal edit