fixing tests

#44