Fix for compile error