Move this to the top of the test-callback, and add a `expect` call.

Otherwise looking good.