Fixing my last stupid update

The last patch broke it, hopefully this one should fix it again.
This is why I shouldn't be allowed nice things!

Tom.