Shield - 2FA for only some users

So I added my file inside \App\Libraries and called it CustomEmail2FA.php.

Inside it is my code:

Code:

<?php

namespace App\Libraries;
use CodeIgniter\Shield\Authentication\Actions\Email2FA;

class CustomEmail2FA extends Email2FA{
    /**
    * Creates an identity for the action of the user.
    *
    * @return string secret
    */
    public function createIdentity(User $user): string
    {
        if($user->email2fa != '1'){
            return '';
        }
        /** @var UserIdentityModel $identityModel */
        $identityModel = model(UserIdentityModel::class);

        // Delete any previous identities for action
        $identityModel->deleteIdentitiesByType($user, $this->type);

        $generator = static fn (): string => random_string('nozero', 6);

        return $identityModel->createCodeIdentity(
            $user,
            [
                'type'  => $this->type,
                'name'  => 'login',
                'extra' => lang('Auth.need2FA'),
            ],
            $generator
        );
    }
}

Then I declared it in Auth.php:

Code:

public array $actions = [
        'register' => \CodeIgniter\Shield\Authentication\Actions\EmailActivator::class,
        'login'    => \App\Libraries\CustomEmail2FA::class,
    ];

Now, when I log in, I always get this error:

Call to a member function getType() on null

From this file:

VENDORPATH/codeigniter4/shield/src/Authentication/Authenticators/Session.php at line 486

Line 486:

Code:

$identity = $this->userIdentityModel->getIdentityByType($this->user, $action->getType());

What am I doing wrong?