Fixed minor bug (YOU CALL THAT MINOR?)