Edit for code consistency