Fixing test break