Copied fix to wk 11 code