Fixed dtype error