Testing code fix