Missing code fix