Fix fun code