Can You Remove Effects On EntityArmorChangeEvent?

Ok So I Want To Setup Where If The Player Puts On A Diffrent Armor Piece except Diamond Armor Its Clears There Effect
Heres My Code There Are No Errors Or Anything It Dosent Do Anything At All

PHP:

       public function ArmorChange(EntityArmorChangeEvent $ev){
        $player = $ev->getEntity();
                     $effect = Effect::getEffect("5");
$effect->setVisible(false);
$effect->setAmplifier("2");
$effect->setDuration("6000");
if($player instanceof Player){
if($ev->getNewItem()->getID() === 310){
$player->addEffect($effect);
            }
        }
        else {
                 $player->removeEffect();
            return;
        }
        }

removeEffect ($effectId) or removeAllEffects ()

Go here

You have to check if it is between 310 and 313, not just 310.