modifying the first test