Temp fix a broken test