Remove external include for mockup-training