updated version and fixed test case