test case bug fixes

Test case was crashing, fix this