Fix 

closed #2