Fixed a small compile error