fixed little bug
fixed another little bug