fix display issues while sampling