Added done() to tests