if statement fixed
if statement fixed