Bug fixes

General bug fixes.