Or you could say "исправляйте" to imply ongoing or repeating action. But in this general context either form works well.