Little update to correct errors