Fix broken arg test