Unit test now works