Justice dept told its budget will not be approved if it does not shape up