Fix compile error in previous commit