[FIX] Add back the deleted files