Fix Unit 2 folder