Bug Fix

fixed a small bug