Fix docblock in test