Fix test :)