Add missing includes to tests