Fix generation specs