Add test coverage for imitation feature