Fixed second test, still getting assertion error