Add missing file for unit test