Fix to last change (marketplace).
Merge pull request #26 from roydude/v1-train

Fix to last change (marketplace).