Fixes custom attribute save