fix tests
Merge pull request #246 from daniellemayne/develop

fix tests