Add this back in, but referencing the updated class name