[tests] Fix tests