[+][FIX][Tests] Fix tests