Fix a bug for correct