test: follow Class name change