Fixed test without `return`