Fixed the os tests as per the comments