Another line of code for the same fix