This conditional logic is moved out of the `cleanup` method and into the `execute!` method.