Possible fix

I'm not 100% sure this is right or not but I'm fairly sure it is.