Fixed minor issue caused by my oversight for plots