This line should be done in both case.