Fix plotting issue