Hook update to enable flag module