Fixes wrong class name