update to use new tests