small fixes that I need