Small define fix