Fixed obvious bug