Bugfix

As it says on the tin