Fix widget test