Correcting class name of return value.