Small update for code consistency