minor bug fix in expected_result