Fix a place that may lead to bugs