Minor Code Fix

Minor Fix