Update docblock with correct types