Added some missing unit tests