Fixed another small issue