Fixing broken test

Just fixing a test that fails...