Fix runtime error

TBR