Fixed minor issue