Better 1 line change to fix