unit test changed to match api changes