Add missing then