bug corrrection