Bug fix (to remove NaN errors)