If statement bugfix