Update Bats test script