needed to change the classname again
Merge branch 'release/0.6.1'