Fixes issue with missing test coverage