Can you make this an if block rather than an if surprise?