Attempting to fix test comparison