Add function to fix