[TEST] Fix tests