fixed more tests bro