Fixed version check and added comment