Change to using updated non-deprecated annotation