fixed broken code
Merge pull request #80 from Petercopter/fixing-tests

fixed broken code