fixed the test to include corner cases