fixed tests broken by my last change