update fixing the bug