Add missing #includes to new test