Solved minor mistake