fixed test
Merge pull request #40 from aidanbebbington/master

fixed test