Bug fix

Update fixes #1