Fix tests
Merge branch 'sonata/master'