update tests to make them pass