Fixed small warning