Add missing thens