Fix type check problem (2)