nicer tests
fixing tests