improve screenshot test