Unit Test Fixes