Fix plotting imports