Fix code in phpunit