Fixing a problem with dtypes introduced in the last commit