It's a little weird to require Mock to run XBlocks. Maybe move this import into the function that uses it.