[Tests] Fix tests (I presume)