Fixing trainTracker