Fix issue within test