Small adjustment to function locking