little BUG fix