Small bug fix for version checked