Fixed failing tests :)