fix(tests): Fixed generate test