Rewrite before_save callback