Add flag to toggle signal injection