Fix docblock issues