Add missing change to fix compile error