Fixed silly bug. Generation should work properly.