Fixed small build issue