Fix action wrapper