remove more plotting commands from tests