Add missing use statement for Query\Criterion