fixed some args stuff