Simple test fix