Correction.
Merge pull request #246 from roydude/v1-train

Correction.