Fix dtype to avoid warning