Unit test now passes