fixed some tests that broke