Fix minor type checking issue