Bug fixes
Update