Adding done() to tests