fixed dropout bug