Add argument to activate testing