Fix dtype test