Now tests passes