updated tests to reflect interface change