This somehow fixes test 2 (check line 372)