small fix, tested