Minor Bug Fix

done