Fix comparison for broken test