Fix due to breaking change in unittest
Merge pull request #1 from sigmundch/master

Fix due to breaking change in unittest