bug fix
demo class