Fix https://github.com/teawater/kgtp/issues/6