Fix(?) tests