Add logic for PHPUnit