[BUG] Fix type mismatch.