Test script fix