Add done callback.

Loading data is removing it is an asynchronous operation.  Thus, in order for
the tests to be reliable, it needs to be treated as such.