fixed tests after change