Fixed broken test. Bad me for checking in broken test.