TST Fix tests