fix the minor issue