Fix dimensions of test case