Fix test bug identified by @evanpx