Added handling of passing in null as AST to schedule, which will happen
if a TB is not matched with any AST