Tiny test fix