Changed to use a more generic SObject in the unit test