Follow class name change
Merge pull request #427 from kou/patch-1

Follow class name change