add #[cfg(test)] attribute