Changed it to inherit from class it is testing