Fix wrong conditional
Fix #6