I don't like forcing a cast. It mean we didn't specified the sme output dtype as numpy. Can you make sure we use the same output dtype and remove this cast?