Patch for test

This patches improve tests.
These are not necessary soon but would be needed when we bump core to later versions.
You could put it maybe in the ```mock``` package?