Fix for comment model addition