fix bug (again...)