change class name

Signed-off-by: Erica <e@energy7.com>