small fix to better results