Changing test to use derived class