Fix for generation menu