Fix Test Bugs