fix to basic test case