little fix
little fix #2