Gear changes will be clunky if you change under full power, particularly in the highest power mode.
I have the same issue on my Kona (Shimano E8000)
Unfortunately it's difficult to avoid as it takes a second for the power to cut off after you stop pedalling, presumably to make the bike...