Fix script for new screenshot directory layout