Improved weight test case