Add missing locks