updated tests to work