Fix. Gotta see why the test was passing though