Fix bug: null error