[CLEANUP] Replace deprecated interface from unittests