Fixed test return type