Fixing a broken test