Fix #336. Add unit test