fixing test break