Modified return value to match the type signature