Fix test to match changes
Merge pull request #438 from ratajczak/master

Fix test to match changes