fixed a broken test due to recent changes