Fix due to breaking change in unittest