Fix compile error from previously merged PR