Add missing files to test