Sorry! Fixing a bug

I missed this one; but I've also added a test to stop it happening again