Solved minor bug