Fix minor compile error