Bug fix

fixes #11