Bug fixes, unit test fixes