Fix slight errors