add missing unit tests for previous commit