modify keygen wrapper