Fix tests docblock