Fix to compile error