small fixes for my last commit