slight fixes