updating for new tests