Fixes misnamed class