Attempted to correct the code block

Didn't work previously.