updatet tests to match the changes