Fix minor issue