Fix issue in "Class Name" (Close #7)

Signed-off-by: Luca Bernardi <luka.bernardi@gmail.com>