Bug fix - if syntax