Fixed so test hopefully works