Fix docblock type