Ao tentar fazer um IF, você está tentando usar uma multi-statement table-valued function (MSTVF) ao invés de uma inline table valued function (TVF).
Por isso a sintaxe muda um pouco e as características mudam também (performance, por exemplo).
Para manter como uma TVF (que geralmente é mais rápida), você poderia escrever: