Test fixes

Test fixes