fixed tests after that last commit