Re-instated the hack
Merge pull request #212 from roydude/v1-train

Re-instated the hack