fixed small issue causing test failure