Fix screenshot return value