Restore incorrectly removed test