Fix args check