Fixed another build error