Add support for extend inside of condition