Improve the OS check