Code snippet fix