fix up test-case, but done is getting called twice now