Updating test for better code coverage