Restore if statement