Fix test
Merge pull request #31 from brahms10/develop

Fix test