I've just realised that the above hack would only work for v2.0.9 (or earlier) of the assessment extension. If you're using a more up-to-date version then I think you'll need to comment out this line as well.