Add missing conditional