Updated test code using new API changes