Updated test case to match product class update