Fix broken test in previous commit