Fix another small issue