small fix, less code