fixing tests
fixing tests