Fix issue causing test to fail