Updated the failing tests