Update tests to fix copy-paste error, and add done()