remove custom class for notification center tests

use blocks instead