fix new tests