Wouldn't it make more sense to handle it in the logic of testing function?