Fixed call to done. That is only needed for unit tests, which this is not.