Updated behat tests

Continuing work done in https://github.com/sonata-project/sandbox/pull/201