Add missing files for tests