amended tests to pass