add test, small correction in code