Fix functionality.