Fixed missing check