Bug fix: unit test example