Fix a bug caused by old version