fix imshow problem