Added missing done() call in test