Moved assertion to correct test.