Changed assert body class to wait