We analyse two models describing disease transmission and control on regular and small-world networks. We use simulations to find a control strategy that minimizes the total cost of an outbreak, thus balancing the costs of disease against that of the preventive treatment. The models are similar in their epidemiological part, but differ in how the removed/recovered individuals are treated. The differences in models affect choice of the strategy only for very cheap treatment and slow spreading disease. However for the combinations of parameters that are important from the epidemiological perspective (high infectiousness and expensive treatment) the models give similar results. Moreover, even where the choice of the strategy is different, the total cost spent on controlling the epidemic is very similar for both models.