Small type fix