Fix for invalid variable name in test