Update the base class to accept `version` parameter