Fixed expected behavior in test