fixed if bug