fix to work, add missing class

Signed-off-by: Hiroshi Miura <miurahr@linux.com>