solved minor bug