Correction to error handler.
Merge pull request #64 from roydude/v1-train

Correction to error handler.