Added the check to the wrong one.