Fixed files missing from build