Fix compile error. doh