revise dropout removal