Fix the test breakage (probably)

#101