small fix #2