Fix to check expected as array