Fix build

@Squadzone Thanks,gonna give it a shot :)