Bug fixes
Test fixes