hopefully works for tests now