Fixed little decorator bug