Test seems to work now