Fixed error in plot