Fixes forgotten class name