Make tests pass upon generation