fix issue 4