The inevitable fix to the last change.
Merge pull request #60 from roydude/v1-train

The inevitable fix to the last change.