Fix the tests for dispatch

Not sure it really works as expected, but at least it pass the tests!