Apply the same fix to another part of the code.