after updating tests to work