Fixes incorrect class name