Resolve issue https://github.com/cjlee112/spnet/issues/68