fix small errors from changing arguments