Fixed another small issue that caused the small issue before to not be fixed