Update PHPUnit logic