Fixing PHPUnit implementation