Fix compile error - oops