Minor code change for readability