solved a little new bug