test cleanup

Cleanup in a test for the recent API change.