really fix screenshot_for method