Fix for a couple of undefined classes in docblocks

These are the 2 that I've come across, but there may be some more.