can you try if it works too with normalize=True?