Minor define fix