Fixed bug in downstream code
Merge pull request #3 from brandwe/master

Fix to downstream code