Fixed 6050.
Merge pull request #36 from roydude/v1-train

Fixed 6050.