Fixing test that broke because we moved it to another assembly