Fix Bats tests