updated tests for better coverage