Is this on a Linux system? NVMe power management has always been hit or miss for consumer SSDs under Linux because the SSD vendors don't write their firmware against the NVMe spec, they write it to work with the Microsoft Windows NVMe driver and any feature Windows doesn't use is liable to be broken. This applies to basically every SSD brand, by the way.