Modified test for better code coverage