Quick fix whlie watching tonights ci run.
Merge pull request #61 from roydude/v1-train

Quick fix whlie watching tonights ci run.