fix remaining code block