Fixing BATS tests