Fix post gen hook