adds missing unit tests