added a different conditional to fix test