Fix a possible bug