Fixed missing check.