slight type fix