Apply fix to other test too.