update proper test for new API