Patch for test

This patches improve tests.
These are not necessary soon but would be needed when we bump core to later versions.
Oh,,, missed.