Fix a quick bug