Another attempt at fixing code block