Changed small if-thing.