Fixing comment controller bug