temporary fix to test
better plotting