Compile error in cube type impl if struct behind conditional

[torsteingrindvik]

I minified an example to roughly the methods foo and bar in the code below.
I'd expect them both to work since they achieve similar things but only bar works.

#[derive(CubeType)]
struct Foo {
    a: f32,
}

#[cube]
impl Foo {
    // mismatched types
    // expected struct `ExpandElementTyped<_>`
    //    found struct `FooExpand`
    pub fn foo(&self, v: u32) -> Foo {
        if v > 0 {
            Foo { a: 1.0 }
        } else {
            Foo { a: 0.0 }
        }
    }

    // ok
    pub fn bar(&self, v: u32) -> Foo {
        let mut foo = Foo { a: 0.0 };

        if v > 0 {
            foo.a = 1.0
        }

        foo
    }
}

I'm guessing this is related to the conditional in the expansion, see image: