Fix passing dtypes to arange()