Fix another build error - Credit to Codeworkx