Fixed small compile bug