Fix another case of the same error