change serializer to kryo due to OOM