Fix screenshot test