Fix so NEAT works again ready for merge with master :)