fixed tests broken by previous change