Fix tests
Merge pull request #7 from wajam/clement-nlb

Fix tests