Fix phpunit tests

Small fix for #202