Make tests run against the correct class.