Adapt screenshot test to new framework