fixed all tests

@adamggg