Replace with custom dialog intead