Fix cellwidget + edit flash out