fixing a broken test