fix bug, add test class