Not sure this is the easiest way to make tests pass.