Fixing code close