TODO: test this fix