[TASK] Replace assertion method to use dedicated method