Yea I'm pretty sure this is the right fix.