fixes for tests
Merge pull request #2 from WestleyArgentum/master

fixes for tests