Fix args in run method