Remove type annotation from fig 2