fix test breakage from merging #172