Update: added test