Minor change to IF