[+][FIX][Test] Fix tests