Add drain test