Fix code for Linux compile