Fix small compile error