Fixed generated tests fail by default due to imports