quick fix for the modification model