Fix plotting