Fix the kryo tests