Bug Fix

Fix Bugs