Changed if statement. (fix?)