Fixed test with inheritance