#2
fixing tests