Simplified the code, as it is a base class.