Fix code for linux