[Kernel] Fix tests