update tests for better code coverage