FIX: update the Widget correctly