bugfix for test breakage