Fix type check.

Fix type check.