Set proper callback for valid super widget states