Fix a bug introduced in previous version