Fix issues introduced in #12

Forgot to remove notification code.