Fix tests OLOL