small fix
change class name
.