Fix build error, missing files