update tests for billing ui changes