Added a if that had been forgotten