Update for API change in 3.3.x