Files missed in build fix